package cn.jlonely.xyhx.utils;

import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import io.jsonwebtoken.security.Keys;
import javax.crypto.SecretKey;
import java.util.Date;
import java.util.Map;

public class JwtUtils {

    // 生成一个符合HS256要求的密钥
    private static final SecretKey signKey = Keys.secretKeyFor(SignatureAlgorithm.HS256);

    private static final long expire = 1000*60*60; // 过期时间，例如1小时

    /**
     * 生成JWT令牌
     *
     * @param claims JWT第二部分负载 payload 中存储的内容
     * @return
     */
    public static String generateJwt(Map<String, Object> claims) {
        String jwt = Jwts.builder()
                .addClaims(claims)
                .signWith(signKey) // 使用生成的密钥进行签名
                .setExpiration(new Date(System.currentTimeMillis() + expire))
                .compact();
        return jwt;
    }

    /**
     * 解析JWT令牌
     *
     * @param jwt JWT令牌
     * @return JWT第二部分负载 payload 中存储的内容
     */
    public static Claims parseJwt(String jwt) {
        return Jwts.parser()
                .setSigningKey(signKey) // 使用生成的密钥进行验证
                .build()
                .parseClaimsJws(jwt)
                .getBody();
    }
}
